package com.qwe.graduation.utils;

import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;

import java.util.Collections;

/**
 * @author lzf
 * @date 2023/1/8 12:52
 * @description CodeGenerator
 * mybatis-plus代码生成器
 */
public class CodeGenerator {

    public static void main(String[] args) {
        generate();
    }
    private static void  generate(){
        FastAutoGenerator.create("jdbc:mysql://localhost:3306/ssmp?characterEncoding=UTF-8", "root", "123456")
            .globalConfig(builder -> {
                builder.author("林卓锋") // 设置作者
                        .enableSwagger() // 开启 swagger 模式
                        .fileOverride() // 覆盖已生成文件
                        .outputDir("D:\\lzf\\软件工程\\毕设\\ls-graduation\\src\\main\\java"); // 指定输出目录
            })
            .packageConfig(builder -> {
                builder.parent("com.qwe.graduation") // 设置父包名
                        .moduleName(null) // 设置父包模块名
                        .pathInfo(Collections.singletonMap(OutputFile.mapperXml, "D:\\lzf\\软件工程\\毕设\\ls-graduation\\src\\main\\java\\com\\qwe\\graduation\\mapper\\xml")); // 设置mapperXml生成路径
            })
            .strategyConfig(builder -> {
                builder.entityBuilder().enableLombok();
                builder.mapperBuilder().enableMapperAnnotation().build(); //lombok注解
                builder.controllerBuilder().enableHyphenStyle()  // 开启驼峰转连字符
                        .enableRestStyle(); // 开启生成@RestController 控制器
                builder.addInclude("video_leave") // 设置需要生成的表名
                        .addTablePrefix("t_", "c_"); // 设置过滤表前缀
            })
//            .templateEngine(new FreemarkerTemplateEngine()) // 使用Freemarker引擎模板，默认的是Velocity引擎模板
            .execute();
    }
}
